model_dir: run/NMTMediumV1

data:
  train_features_file: src-train.txt
  train_labels_file: tgt-train.txt
  eval_features_file: src-val.txt
  eval_labels_file: tgt-val.txt
  source_vocabulary: src-vocab.txt
  target_vocabulary: tgt-vocab.txt
  source_tokenization: tokenizer.yml
  target_tokenization: tokenizer.yml

eval:
  scorers: bleu
  export_on_best: bleu
  max_exports_to_keep: 5
  steps: 1000

train:
  batch_size: 64
  batch_type: examples
  maximum_features_length: 100
  maximum_labels_length: 100
  save_checkpoints_steps: 1000
  keep_checkpoint_max: 5
  #train_steps: 100000

  # One epoch.
  max_step: null
  single_pass: true
